import gensim
from gensim.models import Word2Vec

# 创建句子
sentences = [["I", "love", "AI"], ["AI", "is", "awesome"]]

# 训练Word2Vec模型
model = Word2Vec(sentences, vector_size=100, window=5, min_count=1)
print(model.wv["AI"])